# 作者：位微
# 你不是在写代码而是在解决问题
# 2025年09月17日08时21分19秒
# 2062873410@qq.com
import torch
a=torch.rand(3,3)*10
# print(a.floor_())#向下取整
# print(a.frac())#取小数
# print(a.trunc())#取整数
# print(a.ceil())#向上取整
# print(a.round())#四舍五入
# print(a)
b=torch.rand(3,3)*10
c=torch.randint(0,9,[3,3])
d=torch.randint(0,9,[3,3])
bool_c_eq_d=torch.eq(c,d)
# print(bool_c_eq_d)
a[bool_c_eq_d]=0 #不相等为零
bool_c_gt_d=torch.gt(c,d)
c[bool_c_gt_d]=1
# print(c)
# print(a)
print('*'*100)
print(d)
sorted_d_0=torch.sort(d,dim=0,descending=True)
sorted_d_1=torch.sort(d,dim=1)
print(sorted_d_0)
print(sorted_d_1)
top_2=torch.topk(d,2,dim=0)
print(torch.isinf(d/0))
print(top_2)